#! /usr/bin/env python

# Alunos:   Eliezer de Souza da Silva  RA: 124065
#           Fernanda Brandao Silva     RA: 060727
#           Michel Silva Fornaciali    RA: 071884
# IA368R: Eduardo Valle - Trabalho 2, Programa 2

import sys
import time
from kmeans import *
import numpy as np

#ao rodar este script passar na entrada padrao uma sequencia de pontos 
#separados por linha cada ponto eh tres numeros separados por espaco
#passar um inteiro k como parametro
k=int(sys.argv[1])

points = np.genfromtxt(sys.stdin, delimiter=" ")

codebook, assignments = kmeans(points,k)

print_points(sys.stdout,codebook)
